//
// Created by ZhihuaLiu on 21-1-5.
//

/*
3d_iou (cpu)
3d_nms (cpu)
fast postprocess(CPU)
Written by Zhihua Liu
*/

<%

setup_pybind11(cfg)
%>
#include <pybind11/pybind11.h>
#include <pybind11/numpy.h>
#include <pybind11/stl.h>
#include <stdio.h>
#include <math.h>
#include <vector>
#include<iostream>
#include <iomanip>
#include <cstring>
#include <algorithm>    // std::sort
#include "py_decode.cpp"
//PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
PYBIND11_MODULE(py_decode_api, m) {
        m.def("decode_head_cpu", &decode_head_cpu,"decode single head bbox scores dir pos");
        m.def("nms_cpu_all", &nms_cpu_all,"nms");
}
//cfg['compiler_args'] = ['-std=c++14']
//cfg['sources'] = ['cc_decode.cpp','py_decode.cpp']